/**
* Copyright (C) 2013-2024 Nanjing Pengyun Network Technology Co., Ltd.
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
*     http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/ 

package py.engine;

import java.util.concurrent.Delayed;
import java.util.concurrent.TimeUnit;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public abstract class DelayedTask extends AbstractTask {
  private final Logger logger = LoggerFactory.getLogger(DelayedTask.class);

  private long delayMs;
  private long origin;

  public DelayedTask(int delayMs) {
    this.origin = System.currentTimeMillis();
    this.delayMs = delayMs;
  }

  protected long getOrigin() {
    return origin;
  }

  public void updateDelay(long newDelayMs) {
    long now = System.currentTimeMillis();
    if (now + newDelayMs >= delayMs + origin) {
      delayMs = newDelayMs;
      origin = now;
    }
  }

  @Override
  public long getDelay(TimeUnit unit) {
    return unit.convert(origin + delayMs - System.currentTimeMillis(), TimeUnit.MILLISECONDS);
  }

  @Override
  public final int compareTo(Delayed delayed) {
    logger.trace("this={}, delayed={}", this, delayed);
    if (delayed == null) {
      return 1;
    }

    if (delayed == this) {
      return 0;
    }

    long d = (getDelay(TimeUnit.MILLISECONDS) - delayed.getDelay(TimeUnit.MILLISECONDS));
    return ((d == 0) ? 0 : ((d < 0) ? -1 : 1));
  }

  @Override
  public abstract Result work();

  @Override
  public String toString() {
    return "DelayedTask{" + "origin=" + origin + ", delayMs=" + delayMs + '}';
  }
}
